package org.jsola.hr.service.impl;

import org.jsola.common.DateKit;
import org.jsola.core.Page;
import org.jsola.hr.common.ExampleKit;
import org.jsola.hr.common.secret.util.CoderUtil;
import org.jsola.hr.common.secret.util.SecureUtils;
import org.jsola.hr.constant.HrConstants;
import org.jsola.hr.dao.IEmpSecretKeysDAO;
import org.jsola.hr.dto.EmpSecretKeysAddDTO;
import org.jsola.hr.dto.EmpSecretKeysUpdateDTO;
import org.jsola.hr.entity.EmpSecretKeysDO;
import org.jsola.hr.query.EmpSecretKeysQuery;
import org.jsola.hr.service.IEmpSecretKeysService;
import org.jsola.hr.vo.EmpSecretKeysListVO;
import org.jsola.hr.vo.EmpSecretKeysVO;
import org.jsola.user.core.TokenUser;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import tk.mybatis.mapper.entity.Example;

import java.security.KeyPair;
import java.security.PrivateKey;
import java.security.PublicKey;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author zhr
 */
@Service("hrEmpSecretKeysServiceImpl")
public class EmpSecretKeysServiceImpl implements IEmpSecretKeysService {

    @Autowired
    private IEmpSecretKeysDAO empSecretKeysDAO;

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public EmpSecretKeysVO save(EmpSecretKeysAddDTO empSecretKeysAddDTO, TokenUser tokenUser) {
        // 转DO
        EmpSecretKeysDO empSecretKeysDO = empSecretKeysAddDTO.to(EmpSecretKeysDO.class);
        // 保存
        empSecretKeysDO = save(empSecretKeysDO, tokenUser.getSiteId(), tokenUser.getUserId());
        return empSecretKeysDO.to(EmpSecretKeysVO.class);
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public int update(EmpSecretKeysUpdateDTO empSecretKeysUpdateDTO, TokenUser tokenUser) {
        //转DO
        EmpSecretKeysDO empSecretKeysDO = empSecretKeysUpdateDTO.to(EmpSecretKeysDO.class);
        //根据主键更新，只更新非null值
        return updateByIdSelective(empSecretKeysDO, tokenUser.getSiteId(), tokenUser.getUserId());
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public int deleteByIds(TokenUser tokenUser, String... empSecretKeysIds) {
        return logicalDeleteByIds(tokenUser.getSiteId(), tokenUser.getUserId(), (Object[]) empSecretKeysIds);
    }


    @Override
    public EmpSecretKeysVO selectById(String empSecretKeysId, String siteId) {
        EmpSecretKeysDO empSecretKeysDO = selectDOById(empSecretKeysId, siteId);
        if (empSecretKeysDO == null) {
            return null;
        }
        return empSecretKeysDO.to(EmpSecretKeysVO.class);
    }

    @Override
    public List<EmpSecretKeysListVO> select(EmpSecretKeysQuery empSecretKeysQuery, String siteId) {
        List<EmpSecretKeysDO> empSecretKeysDOList = selectDO(empSecretKeysQuery, siteId);
        if(CollectionUtils.isEmpty(empSecretKeysDOList)) {
            return empSecretKeysDOList == null ? null : new ArrayList<>();
        }
        return empSecretKeysDOList.stream()
                .map(empSecretKeysDO -> empSecretKeysDO.to(EmpSecretKeysListVO.class))
                .collect(Collectors.toList());
    }

    @Override
    public int selectCount(EmpSecretKeysQuery empSecretKeysQuery, String siteId) {
        Example example = buildExample(empSecretKeysQuery,siteId);
        return empSecretKeysDAO.selectCountByExample(example);
    }

    @Override
    public Page<EmpSecretKeysListVO> selectPage(EmpSecretKeysQuery empSecretKeysQuery, String siteId) {
        Example example = buildExample(empSecretKeysQuery,siteId);
        Page<EmpSecretKeysDO> page = empSecretKeysDAO.selectPageByExample(example,
                empSecretKeysQuery.getPageNo(),
                empSecretKeysQuery.getPageSize());

        return page.to(EmpSecretKeysListVO.class);
    }

    @Override
    public EmpSecretKeysDO selectDOById(String empSecretKeysId, String siteId) {
        return listById(empSecretKeysId, siteId);
    }

    @Override
    public List<EmpSecretKeysDO> selectDO(EmpSecretKeysQuery empSecretKeysQuery, String siteId) {
        Example example = buildExample(empSecretKeysQuery,siteId);
        return empSecretKeysDAO.selectByExample(example);
    }


    /**
     * 根据查询参数，构建example

     * @param empSecretKeysQuery 查询参数
     * @param siteId 所属站点id
     * @return example
     */
    private Example buildExample(EmpSecretKeysQuery empSecretKeysQuery, String siteId) {
        Example example = new Example(EmpSecretKeysDO.class);
        example.and()
                .andEqualTo("valid", true)
                .andEqualTo("siteId", siteId);
        if (empSecretKeysQuery.getEmpId() != null) {
            example.and().andLike("empId", "%" + empSecretKeysQuery.getEmpId() + "%");
        }
        if (empSecretKeysQuery.getKeyType() != null) {
            example.and().andLike("keyType", "%" + empSecretKeysQuery.getKeyType() + "%");
        }
        if (empSecretKeysQuery.getKeyStr() != null) {
            example.and().andLike("keyStr", "%" + empSecretKeysQuery.getKeyStr() + "%");
        }
        if (empSecretKeysQuery.getKeySalt() != null) {
            example.and().andLike("keySalt", "%" + empSecretKeysQuery.getKeySalt() + "%");
        }
        if (empSecretKeysQuery.getCompanyId() != null) {
            example.and().andEqualTo("companyId", empSecretKeysQuery.getCompanyId());
        }
        if (empSecretKeysQuery.getGmtCreateStart() != null &&
                empSecretKeysQuery.getGmtCreateEnd() != null) {
            example.and().andBetween("gmtCreate", empSecretKeysQuery.getGmtCreateStart(),
                    empSecretKeysQuery.getGmtCreateEnd());
        }
        if (empSecretKeysQuery.getCreateUserId() != null) {
            example.and().andLike("createUserId", "%" + empSecretKeysQuery.getCreateUserId() + "%");
        }
        if (empSecretKeysQuery.getGmtModifiedStart() != null &&
                empSecretKeysQuery.getGmtModifiedEnd() != null) {
            example.and().andBetween("gmtModified", empSecretKeysQuery.getGmtModifiedStart(),
                    empSecretKeysQuery.getGmtModifiedEnd());
        }
        if (empSecretKeysQuery.getModifiedUserId() != null) {
            example.and().andLike("modifiedUserId", "%" + empSecretKeysQuery.getModifiedUserId() + "%");
        }
        // 排序
        ExampleKit.setExampleOrder(example,empSecretKeysQuery.getOrders());
        return example;
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public String generateEmpRSA(EmpSecretKeysUpdateDTO empSecretKeysUpdateDTO, TokenUser tokenUser) throws Exception {
        Date now = DateKit.getToday();
        //生成秘钥对
        KeyPair keyPair = SecureUtils.genKeyPair(1024);
        //获取公钥，并以base64格式打印出来
        PublicKey publicKey = keyPair.getPublic();
        String pubKey = new String(CoderUtil.encodeBase64(publicKey.getEncoded()));
        //获取私钥，并以base64格式打印出来
        PrivateKey privateKey = keyPair.getPrivate();
        String priKey = new String(CoderUtil.encodeBase64(privateKey.getEncoded()));

        empSecretKeysUpdateDTO.setKeyStr(pubKey);
        EmpSecretKeysDO empSecretKeysParams = new EmpSecretKeysDO();
        empSecretKeysParams.setCompanyId(empSecretKeysUpdateDTO.getCompanyId());
        empSecretKeysParams.setSiteId(tokenUser.getSiteId());
        empSecretKeysParams.setEmpId(empSecretKeysUpdateDTO.getEmpId());
        empSecretKeysParams.setKeyType(empSecretKeysUpdateDTO.getKeyType());
        EmpSecretKeysDO empSecretKeysDO = empSecretKeysDAO.selectOne(empSecretKeysParams);
        if(empSecretKeysDO == null){
            // 添加员工秘钥KEY
            EmpSecretKeysDO empSecretKeysDOAdd = empSecretKeysUpdateDTO.to(EmpSecretKeysDO.class);
            empSecretKeysDOAdd.setValid(true);
            empSecretKeysDOAdd.setSiteId(tokenUser.getSiteId());
            empSecretKeysDOAdd.setCompanyId(empSecretKeysUpdateDTO.getCompanyId());
            empSecretKeysDOAdd.setGmtCreate(now);
            empSecretKeysDOAdd.setCreateUserId(tokenUser.getUserId());
            empSecretKeysDOAdd.setGmtModified(now);
            empSecretKeysDOAdd.setModifiedUserId(tokenUser.getUserId());
            empSecretKeysDAO.insert(empSecretKeysDOAdd);
        }else{
            //更新员工秘钥KEY
            update(empSecretKeysUpdateDTO, tokenUser);
        }
        return priKey;
    }
}




